class Config:
    CONFIG_CLASSES = ('daisy', 'dandelion', 'roses', 'sunflowers', 'tulips')

    CONFIG_BATCH_SIZE = 16
    CONFIG_ROOT_PATH = r'H:/notebook/AlexNet/AlexNet_pycharm/flower_photos/'
    CONFIG_SAVE_PATH = r'./alexnet.pth'
    CONFIG_TEST_IMG = r'1.jpg'
    CONFIG_SAVE_FIG_PATH = r'out.jpg'
